from ConectaMysql import *

class GerenciaModelo(ConectaMysql):    
    def __init__(self):        
        ConectaMysql.__init__(self)

    def getModelos(self,  parametro = "", alterar = False):
        self.__parametro = parametro
        self.__visualizarAlterar = alterar

        self.__selectBase = ("SELECT * FROM tb_modelo " +
                            "INNER JOIN tb_marca ON " +
                            "tb_marca.MARCAid = tb_modelo.MODELO_MARCAid")

       

        self.__selectFiltro1 =  (" WHERE (tb_modelo.MODELOid = '" + self.__parametro + "')") 

        self.__selectFiltro2 = (" or (tb_marca.MARCAnome like " + "'%" + self.__parametro + "%')" +
                                " or (tb_modelo.MODELOcor = '" + self.__parametro + "')" +
                                " or (tb_modelo.MODELOano = '" + self.__parametro + "')" +
                                " or (tb_modelo.MODELOnome like " + "'%" + self.__parametro + "%')")


        if ((parametro <> "") and (self.__visualizarAlterar == True)):
            self.__select = self.__selectBase + self.__selectFiltro1

        elif((parametro <> "") and (self.__visualizarAlterar <> True)):
            self.__select = self.__selectBase + self.__selectFiltro1 + self.__selectFiltro2

        else:
            self.__select = self.__selectBase


        
        self.__result  = self.executaQuery(self.__select)
        
        self.__arrayModelo = []

        for self.__linha in self.__result:
            self.__arrayModelo.append(self.__linha)

        return self.__arrayModelo


    def setModelo(self, objectModelo):
        self.__objectModelo = objectModelo

        insertModelo = self.executaQuery(
            "INSERT INTO tb_Modelo(" +
            "`MODELOnome`," +
            "`MODELOano`," +
            "`MODELOcor`," +
            "`MODELO_MARCAid`" +
            ") VALUES (" +
            "'"+str(self.__objectModelo.getModelo())+ "',"+
            "'"+str(self.__objectModelo.getAno()) + "',"+
            "'"+str(self.__objectModelo.getCor()) + "',"+
            "'"+str(self.__objectModelo.getMarcaId())+"'"
            ")")

        if (insertModelo <> False):
            print "Cadastrado Com Sucesso"

    def alterModelo(self, modeloId, objectModelo):
        self.__objectModelo = objectModelo
        self.__modeloIdAntigo = modeloId
        

        alterModelo = self.executaQuery(
            "UPDATE tb_Modelo SET " +
            "MODELOnome = '" + str(self.__objectModelo.getModelo())+"',"+
            "MODELOano = '" + str(self.__objectModelo.getAno())+"',"+
            "MODELOcor = '" + str(self.__objectModelo.getCor())+"',"+
            "MODELO_MARCAid = '" + str(self.__objectModelo.getMarcaId())+"'"+
            " WHERE (MODELOid = '" + str(self.__modeloIdAntigo) + "')")
        if (alterModelo <> False):
            print "Cadastro alterado com sucesso"


    

